import cv2

# 初始化摄像头
cap = cv2.VideoCapture(1)

if not cap.isOpened():
    print("无法打开摄像头")
    exit()

# 设置照片保存的路径和文件名格式
photo_index = 0

while True:
    # 读取帧
    ret, frame = cap.read()
    
    if not ret:
        print("无法接收帧，退出")
        break
    
    # 显示摄像头的实时画面
    cv2.imshow('Camera', frame)
    
    # 等待按键事件
    key = cv2.waitKey(1)
    
    if key == ord(' '):  # 如果按下空格键
        # 生成照片文件名
        photo_filename = f'./new/photo_{photo_index}.jpg'
        # 保存当前帧为图片
        cv2.imwrite(photo_filename, frame)
        print(f"拍摄并保存图片: {photo_filename}")
        photo_index += 1
    
    # 按下 'q' 键退出
    if key == ord('q'):
        break

# 释放摄像头资源并关闭窗口
cap.release()
cv2.destroyAllWindows()
